from torch.optim import Adam, SGD, SparseAdam, Adagrad, Adadelta, RMSprop